package com.johnny.dataStructures.search;

import java.util.ArrayList;
import java.util.List;

/**
 * 二分查找的前提是该数字有序
 */
public class BinarySearch {
    public static void main(String[] args) {
        int[] arr = {1, 8, 10, 89, 1000,1000, 1234};
//        int resIndex = binarySearch(arr, 0, arr.length - 1, 89);
//        System.out.println("resIndex=" + resIndex);

        List resIndexList = binarySearch2(arr, 0, arr.length - 1, 1000);
        System.out.println("resIndexList=" + resIndexList);
    }

    //二分查找算法

    /**
     * @param arr     数组
     * @param left    左边的索引
     * @param right   右边的索引
     * @param findVal 要查找的值
     * @return 如果找到就返回下标，如果没有找到，就返回-1
     */
    public static int binarySearch(int[] arr, int left, int right, int findVal) {

        //当left>right 时，说明递归这个数字 没找到值
        if (left > right) {
            return -1;
        }
        int mid = (left + right) / 2;
        int midVal = arr[mid];

        if (findVal > midVal) {  //向右递归
            return binarySearch(arr, mid + 1, right, findVal);
        } else if (findVal < midVal) { //向左递归
            return binarySearch(arr, left, mid - 1, findVal);
        } else {
            return mid;
        }
    }


    public static List binarySearch2(int[] arr, int left, int right, int findVal) {

        //当left>right 时，说明递归这个数字 没找到值
        if (left > right) {
            return new ArrayList<Integer>();
        }
        int mid = (left + right) / 2;
        int midVal = arr[mid];

        if (findVal > midVal) {  //向右递归
            return binarySearch2(arr, mid + 1, right, findVal);
        } else if (findVal < midVal) { //向左递归
            return binarySearch2(arr, left, mid - 1, findVal);
        } else {
            /**
             * 思路分析
             * 1.在找到mid索引值，不要马上返回
             * 4.将ArrayList返回
             */
            List<Integer> resIndexList = new ArrayList<Integer>();
            //2.向 mid 索引值的左边所描，将所有满足1000的元素的下标，加入得到集合ArrayList中
            int temp = mid - 1;
            while (true) {
                if (temp < 0 || arr[temp] != findVal) {
                    break;
                }
                resIndexList.add(temp);
                temp -= 1; //temp向左移动
            }
            resIndexList.add(mid);

            // 3.向 mid 索引值的右边所描，将所有满足1000的元素的下标，加入得到集合ArrayList中
            temp = mid + 1;
            while (true) {
                if (temp > arr.length || arr[temp] != findVal) {
                    break;
                }
                resIndexList.add(temp);
                temp += 1; //temp向右移动
            }
            return resIndexList;
        }
    }

}
